HomeSort by relevance Sort by last modified time
    Searched refs:input_shape_tensor (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
fill_triangular.py 129 def _forward_event_shape_tensor(self, input_shape_tensor):
130 batch_shape, d = input_shape_tensor[:-1], input_shape_tensor[-1]
  /external/tensorflow/tensorflow/core/kernels/
strided_slice_op.cc 208 const Tensor& input_shape_tensor = context->input(0); variable
210 context, input_shape_tensor.dims() == 1,
212 input_shape_tensor.shape().DebugString()));
213 if (input_shape_tensor.dtype() == DT_INT32) {
215 context, TensorShapeUtils::MakeShape(input_shape_tensor.vec<int32>(),
217 } else if (input_shape_tensor.dtype() == DT_INT64) {
219 context, TensorShapeUtils::MakeShape(input_shape_tensor.vec<int64>(),
nn_ops_test.cc     [all...]
  /external/tensorflow/tensorflow/python/tpu/
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]]
    [all...]
  /external/tensorflow/tensorflow/python/ops/
nn_ops.py 596 input_shape_tensor = array_ops.shape(inp)
598 [input_shape_tensor[i] for i in spatial_dims])
    [all...]

Completed in 159 milliseconds