Searched defs:in_shape (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbcast_ops.cc43 const TensorShape in_shape = ctx->InputShape(i); variable
44 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(in_shape),
46 in_shape.DebugString()));
92 const TensorShape in_shape = ctx->InputShape(i); variable
93 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(in_shape),
95 in_shape.DebugString()));
100 for (int64 i = 0; i < in_shape.num_elements(); ++i) {
H A Dlrn_ops.cc37 const TensorShape in_shape = ctx->InputShape(0); variable
38 OP_REQUIRES(ctx, in_shape.dims() == 4,
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dcopy_test.cc245 Shape in_shape = ShapeUtil::MakeShapeWithLayout(F32, {0, 0}, {0, 1}); local
247 auto empty = Literal::CreateFromShape(in_shape);
250 auto param0 = builder.Parameter(0, in_shape, "input");
/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_pooling_gpu.cc40 const auto in_shape = tensor_in.shape(); local
H A Dslice_op.cc336 TensorShape in_shape = context->input(0).shape(); local
342 DoesSliceShapeDifferInOnly1D(in_shape, begin, size, &slice_dim)) {
344 (size_t)in_shape.dim_size(1) * in_shape.dim_size(2) *
345 in_shape.dim_size(3),
346 (size_t)in_shape.dim_size(2) * in_shape.dim_size(3),
347 (size_t)in_shape.dim_size(3), (size_t)1};
H A Dfractional_avg_pool_op.cc260 TensorShape in_shape; variable
262 in_shape.AddDim(orig_input_tensor_shape_flat(i));
268 {0}, DataTypeToEnum<double>::v(), in_shape,
320 {0}, 0, in_shape, &in_backprop_tensor));
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dliveness_util_test.cc150 Shape in_shape = ShapeUtil::MakeShape(F32, {8}); local
153 HloInstruction::CreateParameter(0, in_shape, "param0"));
155 HloInstruction::CreateParameter(1, in_shape, "param1"));

Completed in 441 milliseconds