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

  /external/tensorflow/tensorflow/core/kernels/
linalg_ops_common.cc 180 TensorShape output_tensor_shape({});
193 output_tensor_shape = batch_shape;
194 output_tensor_shape.AppendShape(output_matrix_shape);
202 input_idx, output_idx, output_tensor_shape, &out)) {
211 output_idx, output_tensor_shape, &out));
sparse_to_dense_op.cc 89 TensorShape output_tensor_shape; variable
92 &output_tensor_shape));
94 OP_REQUIRES_OK(c, c->allocate_output(0, output_tensor_shape, &output));
  /external/tensorflow/tensorflow/python/ops/parallel_for/
gradients.py 47 output_tensor_shape = output.shape
75 out.set_shape(output_tensor_shape.concatenate(flat_inputs[i].shape))
  /external/tensorflow/tensorflow/core/grappler/optimizers/
constant_folding.h 160 TensorShapeProto* output_tensor_shape, bool* is_single_element_op) const;
constant_folding.cc 2545 TensorShapeProto input_tensor_shape, output_tensor_shape; local
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
conv_op_helpers.cc 410 TensorShape input_tensor_shape, filter_tensor_shape, output_tensor_shape; local
413 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(output_shape, &output_tensor_shape));

Completed in 338 milliseconds