| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference.cc | 205 int64_t operand_index, HloInstruction* dynamic_size) { in DefaultAction()
217 int64_t operand_index, HloInstruction* dynamic_size) { in HandleGetTupleElement()
230 int64_t operand_index, HloInstruction* dynamic_size) { in HandleTuple() argument
240 int64_t operand_index, HloInstruction* dynamic_size) { in HandleBroadcast()
251 HloInstruction* dynamic_size = in HandleCustomCall() local
293 int64_t operand_index, HloInstruction* dynamic_size) { in HandleCustomCall()
348 int64_t operand_index, HloInstruction* dynamic_size) { in HandleSort()
364 int64_t operand_index, HloInstruction* dynamic_size) { in HandlePad()
416 int64_t operand_index, HloInstruction* dynamic_size) { in HandleReduce()
464 HloInstruction* dynamic_size) { in HandleDot()
[all …]
|
| D | dynamic_padder.cc | 174 HloInstruction* dynamic_size = in ReplaceGetSize() local
259 HloInstruction* dynamic_size, in PadWithScalar()
390 HloInstruction* dynamic_size = output_dynamic_dims[output_dim]; in RewriteDynamicReshapeSplitInput() local
602 HloInstruction* dynamic_size = input_dynamic_dims[input_dim]; in RewriteDynamicReshapeCombineInput() local
780 HloInstruction* dynamic_size = in RewriteReverse() local
820 HloInstruction* dynamic_size = in RewriteReverse() local
1272 HloInstruction* dynamic_size = in RewriteDynamicConcat() local
1296 HloInstruction* dynamic_size = nullptr; in RewriteDynamicSort() local
|
| D | dynamic_padder_test.cc | 62 HloInstruction* dynamic_size = in CustomCallDynamicDimensionInference() local
67 HloInstruction* dynamic_size = in CustomCallDynamicDimensionInference() local
503 Literal dynamic_size = LiteralUtil::CreateR0<int32>(2); in XLA_TEST_F() local
608 Literal dynamic_size = LiteralUtil::CreateR0<int32>(1); in XLA_TEST_F() local
709 Literal dynamic_size = LiteralUtil::CreateR0<int32>(2); in XLA_TEST_F() local
|
| D | dynamic_dimension_inference_test.cc | 1261 auto dynamic_size = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
|
| D | hlo_evaluator.cc | 456 HloInstruction* dynamic_size = in HandleGetDimensionSize() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | slice_op.cc | 116 auto dynamic_size = in Compile() local
178 xla::XlaOp dynamic_size = in Compile() local
|
| D | sequence_ops.cc | 126 auto dynamic_size = (xla::Abs(limit - start) + xla::Abs(delta) - in Compile() local
132 auto dynamic_size = (xla::Ceil(xla::Abs((limit - start) / delta))); in Compile() local
|
| D | tensor_array_ops.cc | 138 bool dynamic_size; in TensorArrayOp() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | tensor_array_ops.py | 58 dynamic_size=None, argument
407 dynamic_size=None, argument
670 dynamic_size=None, argument
1019 dynamic_size=None, argument
1118 def dynamic_size(self): member in TensorArray
1354 dynamic_size=False, 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 | 1085 bool dynamic_size = tensor_array->HasDynamicSize(); in Compute() local
1267 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 | 811 int64_t dynamic_size = GetDynamicSize(i); in Broadcast() local
917 int64_t 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
|