| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference.cc | 81 int64 operand_index, HloInstruction* dynamic_size)>;
101 int64 operand_index, HloInstruction* dynamic_size) { in DefaultAction() argument
113 int64 operand_index, HloInstruction* dynamic_size) { in HandleGetTupleElement() argument
117 parent_->SetDynamicSize(hlo, new_index, dimension, dynamic_size); in HandleGetTupleElement()
126 int64 operand_index, HloInstruction* dynamic_size) { in HandleTuple() argument
128 parent_->SetDynamicSize(hlo, index, dimension, dynamic_size); in HandleTuple()
136 int64 operand_index, HloInstruction* dynamic_size) { in HandleBroadcast() argument
138 parent_->SetDynamicSize(hlo, {}, broadcast_dim, dynamic_size); in HandleBroadcast()
146 int64 operand_index, HloInstruction* dynamic_size) { in HandlePad() argument
156 parent_->SetDynamicSize(hlo, {}, dimension, dynamic_size); in HandlePad()
[all …]
|
| D | hlo_get_dimension_size_rewriter.cc | 44 HloInstruction* dynamic_size = in ReplaceGetSize() local
46 if (dynamic_size != nullptr) { in ReplaceGetSize()
47 TF_RETURN_IF_ERROR(instr->ReplaceAllUsesWith(dynamic_size)); in ReplaceGetSize()
|
| D | dynamic_padder.cc | 123 HloInstruction* dynamic_size = in Run() local
125 if (dynamic_size == nullptr) { in Run()
162 mask_shape, dynamic_size, {})); in Run()
|
| D | hlo_evaluator.cc | 401 HloInstruction* dynamic_size = in HandleGetDimensionSize() local
403 if (dynamic_size != nullptr) { in HandleGetDimensionSize()
405 GetEvaluatedLiteralFor(dynamic_size).Clone(); in HandleGetDimensionSize()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | tensor_array_ops.py | 52 dynamic_size=None, argument
108 if handle is not None and dynamic_size is not None:
118 dynamic_size = dynamic_size or False
158 dynamic_size=dynamic_size,
421 dynamic_size=None, argument
460 self._dynamic_size = dynamic_size
683 dynamic_size=None, argument
737 self._dynamic_size = dynamic_size or False
970 dynamic_size=None, argument
1030 dynamic_size=dynamic_size,
[all …]
|
| D | map_fn.py | 226 dynamic_size=False,
238 dynamic_size=False,
|
| D | functional_ops.py | 101 dtype=elem.dtype, size=n, dynamic_size=False,
210 dtype=elem.dtype, size=n, dynamic_size=False,
407 dynamic_size=False,
428 dynamic_size=False,
|
| D | rnn.py | 1129 dynamic_size=True,
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | decoder.py | 365 dynamic_size = maximum_iterations is None or not is_xla
370 size=0 if dynamic_size else maximum_iterations,
371 dynamic_size=dynamic_size,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensor_array_ops_test.py | 384 dynamic_size=True,
878 dynamic_size=True)
922 def _testWhileLoopWritePackGradients(self, dynamic_size, dtype): argument
929 size=0 if dynamic_size else 3,
930 dynamic_size=dynamic_size)
1010 dynamic_size=False, dtype=dtypes.float32)
1018 dynamic_size=True, dtype=dtypes.float32)
1209 dynamic_size=True,
1247 dynamic_size=True,
1258 dynamic_size=True,
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_array_ops.cc | 138 bool dynamic_size; in TensorArrayOp() local
139 OP_REQUIRES_OK(ctx, ctx->GetAttr("dynamic_size", &dynamic_size)); in TensorArrayOp()
141 ctx, !dynamic_size, in TensorArrayOp()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-tensor-array.pbtxt | 19 …argspec: "args=[\'self\', \'dtype\', \'size\', \'dynamic_size\', \'clear_after_read\', \'tensor_ar…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-tensor-array.pbtxt | 19 …argspec: "args=[\'self\', \'dtype\', \'size\', \'dynamic_size\', \'clear_after_read\', \'tensor_ar…
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TensorArrayV3.pbtxt | 39 name: "dynamic_size"
|
| /external/tensorflow/tensorflow/python/autograph/utils/ |
| D | tensor_list_test.py | 48 l = tensor_array_ops.TensorArray(dtypes.int32, size=0, dynamic_size=True)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_array.h | 142 bool identical_element_shapes, bool dynamic_size, in TensorArray() argument
149 dynamic_size_(dynamic_size), in TensorArray()
|
| D | tensor_array_ops.cc | 1099 bool dynamic_size = tensor_array->HasDynamicSize(); in Compute() local
1102 if (dynamic_size && array_size < max_index + 1) { in Compute()
1284 bool dynamic_size = tensor_array->HasDynamicSize(); in Compute() local
1292 if (dynamic_size && array_size < num_tensors) { in Compute()
|
| /external/tensorflow/tensorflow/contrib/integrate/python/ops/ |
| D | odes.py | 396 t.dtype, size=0, dynamic_size=True),
398 rtol.dtype, size=0, dynamic_size=True))
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | data_structures_test.py | 111 l = tensor_array_ops.TensorArray(dtypes.int32, size=0, dynamic_size=True)
|
| D | data_structures.py | 99 dynamic_size=True,
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_map_ops.py | 251 dtype=t, dynamic_size=False, infer_shape=infer_shape, size=n)
|
| /external/tensorflow/tensorflow/contrib/autograph/examples/benchmarks/ |
| D | cartpole_benchmark.py | 109 tf.TensorArray(tf.float32, 0, dynamic_size=True) for _ in var_list
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | model.py | 784 dtype=dtype, size=window_size, dynamic_size=False)
|
| D | input_pipeline.py | 412 dtype=dtypes.string, size=0, dynamic_size=True)])
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 2376 dynamic_size = pfor_input.get_attr("dynamic_size")
2386 dynamic_size=dynamic_size,
|