Home
last modified time | relevance | path

Searched refs:input_shape_tensor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstrided_slice_op.cc214 const Tensor& input_shape_tensor = context->input(0); in Compute() local
216 context, input_shape_tensor.dims() == 1, in Compute()
218 input_shape_tensor.shape().DebugString())); in Compute()
219 if (input_shape_tensor.dtype() == DT_INT32) { in Compute()
221 context, TensorShapeUtils::MakeShape(input_shape_tensor.vec<int32>(), in Compute()
223 } else if (input_shape_tensor.dtype() == DT_INT64) { in Compute()
226 input_shape_tensor.vec<int64_t>(), &input_shape)); in Compute()
H A Dnn_ops_test.cc963 Tensor input_shape_tensor(DT_INT32, shape2); in BM_AvgPoolBk() local
966 input_shape_tensor.flat<int32>()(i) = input_dims[i]; in BM_AvgPoolBk()
968 inputs.push_back({nullptr, &input_shape_tensor}); in BM_AvgPoolBk()
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu.py1096 input_shape_tensor = input_shape_tensors[idx][core_idx]
1112 math_ops.cast(input_shape_tensor[i], dtypes.int32))
1131 padding = [0, max_dim_size - input_shape_tensor[i]]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlower_tf.cc940 auto input_shape_tensor = rewriter.create<ConstOp>( in matchAndRewrite() local
948 rewriter.create<AddV2Op>(loc, paddings_sum, input_shape_tensor); in matchAndRewrite()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_ops.py863 input_shape_tensor = array_ops.shape(inp)
865 [input_shape_tensor[i] for i in spatial_dims])