Searched refs:input_shape_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 211 const Tensor& input_shape_tensor = context->input(0); in Compute() local
213 context, input_shape_tensor.dims() == 1, in Compute()
215 input_shape_tensor.shape().DebugString())); in Compute()
216 if (input_shape_tensor.dtype() == DT_INT32) { in Compute()
218 context, TensorShapeUtils::MakeShape(input_shape_tensor.vec<int32>(), in Compute()
220 } else if (input_shape_tensor.dtype() == DT_INT64) { in Compute()
222 context, TensorShapeUtils::MakeShape(input_shape_tensor.vec<int64>(), in Compute()
|
| D | nn_ops_test.cc | 948 Tensor input_shape_tensor(DT_INT32, shape2); in BM_AvgPoolBk() local
951 input_shape_tensor.flat<int32>()(i) = input_dims[i]; in BM_AvgPoolBk()
953 inputs.push_back({nullptr, &input_shape_tensor}); in BM_AvgPoolBk()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 1062 input_shape_tensor = input_shape_tensors[idx][core_idx]
1078 math_ops.cast(input_shape_tensor[i], dtypes.int32))
1097 padding = [0, max_dim_size - input_shape_tensor[i]]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 910 auto input_shape_tensor = rewriter.create<ConstOp>( in matchAndRewrite() local
918 rewriter.create<AddV2Op>(loc, paddings_sum, input_shape_tensor); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 703 input_shape_tensor = array_ops.shape(inp)
705 [input_shape_tensor[i] for i in spatial_dims])
|