| /external/tensorflow/tensorflow/core/util/ |
| strided_slice_op.h | 32 // Note that when input_shape is not fully specified, only <final_shape> and
39 // made to set <processing_shape> and <final_shape> based on <input_shape>, but
40 // some dimensions of <processing_shape> and/or <final_shape> may be unknown
48 PartialTensorShape* processing_shape, PartialTensorShape* final_shape,
59 TensorShape* final_shape, bool* is_identity, bool* is_simple_slice,
|
| strided_slice_op.cc | 152 PartialTensorShape* processing_shape, PartialTensorShape* final_shape,
359 final_shape->Clear();
362 final_shape->AddDim(processing_shape->dim_size(gather_index));
364 final_shape->AddDim(1);
375 TensorShape* final_shape, bool* is_identity, bool* is_simple_slice,
388 !partial_final_shape.AsTensorShape(final_shape)) {
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| strided_slice_op.cc | 47 TensorShape final_shape; variable
71 shrink_axis_mask_, &dummy_processing_shape, &final_shape,
99 slice = ctx->builder()->Reshape(slice, final_shape.dim_sizes());
127 TensorShape processing_shape, final_shape; variable
153 shrink_axis_mask_, &processing_shape, &final_shape, &dummy,
159 ctx, final_shape == dy_shape,
161 " instead of ", final_shape.DebugString()));
238 TensorShape final_shape; variable
268 shrink_axis_mask_, &dummy_processing_shape, &final_shape,
271 if (final_shape.num_elements() == 0 && rhs_shape.num_elements() == 0)
[all...] |
| reduction_ops_common.cc | 87 std::vector<int64> final_shape; local
92 final_shape.push_back(dim);
97 final_shape.push_back(1);
129 result = ctx->builder()->Reshape(finalized, final_shape);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| strided_slice_op.cc | 91 TensorShape processing_shape, final_shape; variable
104 &processing_shape, &final_shape, &is_identity,
112 CHECK(tmp.CopyFrom(input, final_shape));
122 CHECK(tmp.CopyFrom(input.Slice(begin[0], end[0]), final_shape));
128 OP_REQUIRES_OK(context, context->allocate_output(0, final_shape, &result));
136 // final_shape being 2D. This isn't strictly necessary, but I don't
141 final_shape.dims() == 2) {
191 TensorShape processing_shape, final_shape; variable
222 shrink_axis_mask, &processing_shape, &final_shape, &is_identity,
228 context, final_shape == dy_shape
283 TensorShape processing_shape, final_shape; variable
[all...] |
| strided_slice_op_test.cc | 109 TensorShape processing_shape, final_shape; local
120 new_axis_mask, shrink_axis_mask, &processing_shape, &final_shape,
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| sparse_tensor.h | 431 ShapeArray final_shape(tensors[0].shape().begin(), tensors[0].shape().end());
432 final_shape[primary_dim] = 0; // We'll build this up as we go along.
447 CHECK_EQ(final_shape[cdim], st_shape[cdim])
451 << ". Expecting shape like: [" << str_util::Join(final_shape, ",")
456 final_shape[primary_dim] =
457 (final_shape[primary_dim] + st_shape[primary_dim]);
464 final_order = UndefinedOrder(final_shape);
491 return SparseTensor(output_ix, output_vals, final_shape, final_order);
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| dirichlet_multinomial.py | 267 final_shape = array_ops.concat([[n], self.batch_shape_tensor(), [k]], 0)
268 x = array_ops.reshape(x, final_shape)
|
| multinomial.py | 266 final_shape = array_ops.concat([[n], self.batch_shape_tensor(), [k]], 0)
267 x = array_ops.reshape(x, final_shape) # [n, B1, B2,..., Bm, k]
|
| util.py | [all...] |
| transformed_distribution.py | 407 final_shape = array_ops.concat([sample_shape, batch_event_shape], 0)
408 x = array_ops.reshape(x, final_shape)
|
| distribution.py | 670 final_shape = array_ops.concat([sample_shape, batch_event_shape], 0)
671 samples = array_ops.reshape(samples, final_shape)
[all...] |
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| conditional_transformed_distribution.py | 89 final_shape = array_ops.concat([sample_shape, batch_event_shape], 0)
90 x = array_ops.reshape(x, final_shape)
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| beam_search_decoder.py | 808 final_shape = array_ops.shape(gather_from)[:1 + len(gather_shape)]
813 output = array_ops.reshape(output, final_shape, name="output")
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| single_machine_test.cc | 165 auto final_shape = ops::Shape(root.WithOpName("final_shape"), reshaped); local
170 ops::Equal(root.WithOpName("valid"), final_shape, expected_shape);
175 {final_shape.output});
206 "new_shape", "reshaped", "final_shape", "expected_shape",
|
| /external/tensorflow/tensorflow/contrib/kfac/python/kernel_tests/ |
| fisher_factors_test.py | 447 final_shape,
455 self.assertEqual(final_shape, cov.get_shape().as_list())
|
| /external/tensorflow/tensorflow/core/ops/ |
| array_ops.cc | [all...] |
