| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 74 out_shape.AppendShape(input_shape_pre_axis); in XlaGather()
75 out_shape.AppendShape(indices_shape_no_index_vectors); in XlaGather()
76 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()
|
| D | tensor_list_ops.cc | 81 list_shape.AppendShape(element_shape); in CreateZerosList()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_resource.cc | 138 ta_shape.AppendShape(shape_); in SetZeroValue()
146 ta_shape.AppendShape(shape_); in SetZeroValue()
171 ta_shape.AppendShape(shape_); in GetOrCreateTensorArrayGradient()
|
| D | xla_compiler.cc | 659 shape.AppendShape(absl::get<TensorShape>(arg.shape)); in XLAShapeForArgument()
677 shape.AppendShape(absl::get<TensorShape>(arg.shape)); in XLAShapeForArgument()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | lookup_interface.cc | 55 expected_value_shape.AppendShape(value_shape()); in CheckKeyAndValueTensorsHelper()
|
| D | tensor_shape_test.cc | 170 s2.AppendShape(s); in TEST()
246 void AppendShape(const TensorShapeOld& shape);
413 void TensorShapeOld::AppendShape(const TensorShapeOld& shape) { in AppendShape() function in tensorflow::__anone850e9110111::TensorShapeOld
|
| D | tensor_shape.h | 192 void AppendShape(const TensorShapeBase& shape);
|
| D | tensor_shape.cc | 383 void TensorShapeBase<Shape>::AppendShape(const TensorShapeBase& shape) { in AppendShape() function in tensorflow::TensorShapeBase
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_array.cc | 104 tensors_[i].shape.AppendShape(rhs->tensors_[i].shape); in CopyShapesFrom()
|
| D | lookup_table_op.cc | 390 expected_shape.AppendShape(key_shape_); in Find()
457 expected_shape.AppendShape(key_shape_); in Insert()
482 expected_shape.AppendShape(key_shape_); in Remove()
543 expected_value_shape.AppendShape(value_shape); in CheckKeyAndValueTensorsForImport()
805 output_shape.AppendShape(table->value_shape()); in Compute()
|
| D | queue_base.h | 121 shape.AppendShape(component_shapes_[i]); in ManyOutShape()
|
| D | determinant_op.cc | 153 out_shape.AppendShape(TensorShape({})); in ComputeAsync()
293 out_shape.AppendShape(TensorShape({})); in ComputeAsync()
|
| D | gather_op.cc | 96 result_shape.AppendShape(indices.shape()); in Compute()
|
| D | lu_op.cc | 87 input_matrix_shape.AppendShape({num_rows, num_cols}); in Compute()
|
| D | random_poisson_op.cc | 296 samples_shape.AppendShape(rate_t.shape()); in Compute()
|
| D | linalg_ops_common.cc | 194 output_tensor_shape.AppendShape(output_matrix_shape); in PrepareOutputs()
|
| D | random_op.cc | 297 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/data/ |
| D | batch_dataset_op.cc | 183 batch_component_shape.AppendShape(first_element.shape()); in GetNextInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | sliding_window_dataset_op.cc | 206 batch_component_shape.AppendShape(first_element.shape()); in GetNextInternal()
|
| D | dense_to_sparse_batch_dataset_op.cc | 91 output_shape.AppendShape(row_shape_); in Dataset()
|
| D | map_and_batch_dataset_op.cc | 534 component_shape.AppendShape(return_values->at(i).shape()); in EnsureOutputAllocated()
|
| D | numa_map_and_batch_dataset_op.cc | 955 component_shape.AppendShape(map_fn_outputs.at(i).shape()); in AllocateOutput()
|