Searched defs:in_shape (Results 1 - 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | bcast_ops.cc | 43 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 D | lrn_ops.cc | 37 const TensorShape in_shape = ctx->InputShape(0); variable 38 OP_REQUIRES(ctx, in_shape.dims() == 4,
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | copy_test.cc | 245 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 D | cudnn_pooling_gpu.cc | 40 const auto in_shape = tensor_in.shape(); local
|
H A D | slice_op.cc | 336 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 D | fractional_avg_pool_op.cc | 260 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 D | liveness_util_test.cc | 150 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