Searched defs:dynamic_size (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference.cc | 101 int64 operand_index, HloInstruction* dynamic_size) { in DefaultAction()
113 int64 operand_index, HloInstruction* dynamic_size) { in HandleGetTupleElement()
126 int64 operand_index, HloInstruction* dynamic_size) { in HandleTuple() argument
136 int64 operand_index, HloInstruction* dynamic_size) { in HandleBroadcast()
146 int64 operand_index, HloInstruction* dynamic_size) { in HandlePad()
169 int64 operand_index, HloInstruction* dynamic_size) { in HandleReduce()
203 int64 operand_index, HloInstruction* dynamic_size) { in HandleDot()
254 int64 operand_index, HloInstruction* dynamic_size) { in HandleTranspose()
265 int64 operand_index, HloInstruction* dynamic_size) { in HandleConvolution()
310 int64 operand_index, HloInstruction* dynamic_size) { in PassThroughDynamicDimension()
[all …]
|
| D | hlo_get_dimension_size_rewriter.cc | 44 HloInstruction* dynamic_size = in ReplaceGetSize() local
|
| D | dynamic_padder.cc | 123 HloInstruction* dynamic_size = in Run() local
|
| D | hlo_evaluator.cc | 401 HloInstruction* dynamic_size = in HandleGetDimensionSize() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | tensor_array_ops.py | 52 dynamic_size=None, argument
421 dynamic_size=None, argument
683 dynamic_size=None, argument
970 dynamic_size=None, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_array_ops.cc | 1099 bool dynamic_size = tensor_array->HasDynamicSize(); in Compute() local
1284 bool dynamic_size = tensor_array->HasDynamicSize(); in Compute() local
|
| D | tensor_array.h | 142 bool identical_element_shapes, bool dynamic_size, in TensorArray()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_array_ops.cc | 138 bool dynamic_size; in TensorArrayOp() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensor_array_ops_test.py | 922 def _testWhileLoopWritePackGradients(self, dynamic_size, dtype): argument
|