| /external/tensorflow/tensorflow/core/framework/ |
| D | lookup_interface.cc | 55 expected_value_shape.AppendShape(value_shape()); in CheckKeyAndValueTensorsHelper()
90 fullsize_value_shape.AppendShape(value_shape()); in CheckFindArguments()
|
| D | tensor_shape_test.cc | 171 s2.AppendShape(s); in TEST()
269 void AppendShape(const TensorShapeOld& shape);
436 void TensorShapeOld::AppendShape(const TensorShapeOld& shape) { in AppendShape() function in tensorflow::__anon127dd62f0111::TensorShapeOld
|
| D | tensor_shape.h | 213 void AppendShape(const TensorShapeBase& shape);
|
| D | tensor_shape.cc | 488 void TensorShapeBase<Shape>::AppendShape(const TensorShapeBase& shape) { in AppendShape() function in tensorflow::TensorShapeBase
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_resource.cc | 139 ta_shape.AppendShape(shape_); in SetZeroValue()
147 ta_shape.AppendShape(shape_); in SetZeroValue()
172 ta_shape.AppendShape(shape_); in GetOrCreateTensorArrayGradient()
|
| D | xla_compiler.cc | 894 shape.AppendShape(absl::get<TensorShape>(arg.shape)); in XLAShapeForArgument()
912 shape.AppendShape(absl::get<TensorShape>(arg.shape)); in XLAShapeForArgument()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 80 out_shape.AppendShape(input_shape_pre_axis); in XlaGather()
81 out_shape.AppendShape(indices_shape_no_index_vectors); in XlaGather()
82 out_shape.AppendShape(input_shape_post_axis); in XlaGather()
|
| D | tensor_array_ops.cc | 81 ta_shape.AppendShape(elem_shape); in MaybeInitializeTensorArray()
164 ta_shape.AppendShape(shape); in Compile()
518 ta_shape.AppendShape(elem_shape); in Compile()
|
| D | stack_ops.cc | 72 stack_shape.AppendShape(elem_shape); in MaybeInitializeStack()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_array.cc | 103 tensors_[i].shape.AppendShape(rhs->tensors_[i].shape); in CopyShapesFrom()
|
| D | queue_base.h | 117 shape.AppendShape(component_shapes_[i]); in ManyOutShape()
|
| D | lookup_table_op.cc | 412 expected_shape.AppendShape(key_shape_); in Find()
479 expected_shape.AppendShape(key_shape_); in Insert()
504 expected_shape.AppendShape(key_shape_); in Remove()
565 expected_value_shape.AppendShape(value_shape); in CheckKeyAndValueTensorsForImport()
845 output_shape.AppendShape(table->value_shape()); in Compute()
|
| D | random_poisson_op.cc | 300 samples_shape.AppendShape(rate_t.shape()); in Compute()
|
| D | random_op.cc | 169 samples_shape.AppendShape(alpha_t.shape()); in Compute()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | ops.cc | 90 shape.AppendShape(TensorShape{first.tensor.shape()}); in Initializer()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | determinant_op.cc | 153 out_shape.AppendShape(TensorShape({})); in ComputeAsync()
293 out_shape.AppendShape(TensorShape({})); in ComputeAsync()
|
| D | lu_op.cc | 87 input_matrix_shape.AppendShape({num_rows, num_cols}); in Compute()
|
| D | linalg_ops_common.cc | 196 output_tensor_shape.AppendShape(output_matrix_shape); in PrepareOutputs()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 245 output_shape.AppendShape({num_rows, num_cols}); in AllocateOutput()
250 output_transposed_shape.AppendShape({num_rows, num_cols}); in AllocateOutput()
251 output_shape.AppendShape({num_cols, num_rows}); in AllocateOutput()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | sliding_window_dataset_op.cc | 215 batch_component_shape.AppendShape(first_element.shape()); in GetNextInternal()
|
| D | dense_to_sparse_batch_dataset_op.cc | 89 output_shape.AppendShape(row_shape_); in Dataset()
|
| D | rebatch_dataset_op.cc | 499 component_shape.AppendShape(remaining_shape); in GetNextInternal()
|
| D | map_and_batch_dataset_op.cc | 509 component_shape.AppendShape(return_values->at(i).shape()); in EnsureOutputAllocated()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils.cc | 861 batch_component_shape.AppendShape(first_element_shape); in CopyBatch()
|