| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference.cc | 203 int64 operand_index, HloInstruction* dynamic_size) { in DefaultAction()
215 int64 operand_index, HloInstruction* dynamic_size) { in HandleGetTupleElement()
228 int64 operand_index, HloInstruction* dynamic_size) { in HandleTuple() argument
238 int64 operand_index, HloInstruction* dynamic_size) { in HandleBroadcast()
249 HloInstruction* dynamic_size = in HandleCustomCall() local
291 int64 operand_index, HloInstruction* dynamic_size) { in HandleCustomCall()
346 int64 operand_index, HloInstruction* dynamic_size) { in HandleSort()
362 int64 operand_index, HloInstruction* dynamic_size) { in HandlePad()
414 int64 operand_index, HloInstruction* dynamic_size) { in HandleReduce()
463 HloInstruction* dynamic_size) { in HandleDot()
[all …]
|
| D | dynamic_padder_test.cc | 61 HloInstruction* dynamic_size = in CustomCallDynamicDimensionInference() local
66 HloInstruction* dynamic_size = in CustomCallDynamicDimensionInference() local
430 Literal dynamic_size = LiteralUtil::CreateR0<int32>(2); in XLA_TEST_F() local
494 Literal dynamic_size = LiteralUtil::CreateR0<int32>(1); in XLA_TEST_F() local
595 Literal dynamic_size = LiteralUtil::CreateR0<int32>(2); in XLA_TEST_F() local
|
| D | dynamic_padder.cc | 170 HloInstruction* dynamic_size = in ReplaceGetSize() local
249 HloInstruction* dynamic_size, in PadWithScalar()
380 HloInstruction* dynamic_size = output_dynamic_dims[output_dim]; in RewriteDynamicReshapeSplitInput() local
592 HloInstruction* dynamic_size = input_dynamic_dims[input_dim]; in RewriteDynamicReshapeCombineInput() local
1171 HloInstruction* dynamic_size = in RewriteDynamicConcat() local
1195 HloInstruction* dynamic_size = nullptr; in RewriteDynamicSort() local
|
| D | dynamic_dimension_inference_test.cc | 1261 auto dynamic_size = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
|
| D | hlo_evaluator.cc | 421 HloInstruction* dynamic_size = in HandleGetDimensionSize() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | slice_op.cc | 105 std::vector<int64> dynamic_size; in Compile() local
113 auto dynamic_size = in Compile() local
168 auto dynamic_size = in Compile() local
|
| D | tensor_array_ops.cc | 138 bool dynamic_size; in TensorArrayOp() local
|
| D | strided_slice_op.cc | 568 std::vector<int64> dynamic_size; in Compile() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | tensor_array_ops.py | 59 dynamic_size=None, argument
400 dynamic_size=None, argument
657 dynamic_size=None, argument
1005 dynamic_size=None, argument
1104 def dynamic_size(self): member in TensorArray
1325 dynamic_size=False, infer_shape=True): argument
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure.py | 56 def _TensorArrayStructure(dtype, element_shape, dynamic_size, infer_shape): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_array_ops.cc | 1092 bool dynamic_size = tensor_array->HasDynamicSize(); in Compute() local
1276 bool dynamic_size = tensor_array->HasDynamicSize(); in Compute() local
|
| D | tensor_array.h | 140 bool identical_element_shapes, bool dynamic_size, in TensorArray()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensor_array_ops_test.py | 972 def _testWhileLoopWritePackGradients(self, dynamic_size, dtype): argument
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 797 int64 dynamic_size = GetDynamicSize(i); in Broadcast() local
903 int64 dynamic_size = GetDynamicSize(dnum) - start_indices[dnum]; in SliceInternal() local
|
| /external/mesa3d/src/intel/vulkan/ |
| D | genX_cmd_buffer.c | 3480 bool dynamic_size = cmd_buffer->state.gfx.dynamic.dyn_vbo_size; in genX() local
|