Searched refs:input_shape_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 208 const Tensor& input_shape_tensor = context->input(0); in Compute() local
210 context, input_shape_tensor.dims() == 1, in Compute()
212 input_shape_tensor.shape().DebugString())); in Compute()
213 if (input_shape_tensor.dtype() == DT_INT32) { in Compute()
215 context, TensorShapeUtils::MakeShape(input_shape_tensor.vec<int32>(), in Compute()
217 } else if (input_shape_tensor.dtype() == DT_INT64) { in Compute()
219 context, TensorShapeUtils::MakeShape(input_shape_tensor.vec<int64>(), in Compute()
|
| D | nn_ops_test.cc | 925 Tensor input_shape_tensor(DT_INT32, shape2); in BM_AvgPoolBk() local
928 input_shape_tensor.flat<int32>()(i) = input_dims[i]; in BM_AvgPoolBk()
930 inputs.push_back({nullptr, &input_shape_tensor}); in BM_AvgPoolBk()
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | fill_triangular.py | 129 def _forward_event_shape_tensor(self, input_shape_tensor): argument
130 batch_shape, d = input_shape_tensor[:-1], input_shape_tensor[-1]
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 611 input_shape_tensor = input_shape_tensors[idx][core_idx]
634 math_ops.cast(input_shape_tensor[i], dtypes.uint32))
644 padding = [0, s.value - input_shape_tensor[i]]
648 padding = [0, maximum_shapes[idx][i] - input_shape_tensor[i]]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 596 input_shape_tensor = array_ops.shape(inp)
598 [input_shape_tensor[i] for i in spatial_dims])
|