| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 152 xla::ValueInferenceMode mode = xla::ValueInferenceMode::kValue);
155 xla::ValueInferenceMode mode = xla::ValueInferenceMode::kValue);
160 xla::ValueInferenceMode mode = xla::ValueInferenceMode::kValue);
163 xla::ValueInferenceMode mode = xla::ValueInferenceMode::kValue);
168 xla::ValueInferenceMode mode = xla::ValueInferenceMode::kValue);
173 xla::ValueInferenceMode mode = xla::ValueInferenceMode::kValue);
176 xla::ValueInferenceMode mode = xla::ValueInferenceMode::kValue);
182 xla::ValueInferenceMode mode = xla::ValueInferenceMode::kValue);
185 xla::ValueInferenceMode mode = xla::ValueInferenceMode::kValue);
190 xla::ValueInferenceMode mode = xla::ValueInferenceMode::kValue);
[all …]
|
| D | xla_op_kernel.cc | 141 xla::ValueInferenceMode mode) { in ConstantInput()
180 xla::ValueInferenceMode mode) { in ConstantInput()
187 xla::Literal* constant_literal, xla::ValueInferenceMode mode) { in ConstantInputReshaped()
235 int index, int64_t* out, xla::ValueInferenceMode mode) { in ConstantInputAsIntScalar()
242 absl::string_view name, int64_t* out, xla::ValueInferenceMode mode) { in ConstantInputAsIntScalar()
248 int index, double* out, xla::ValueInferenceMode mode) { in ConstantInputAsFloatScalar()
377 int index, std::vector<int64_t>* out, xla::ValueInferenceMode mode) { in ConstantInputAsIntVector()
385 xla::ValueInferenceMode mode) { in ConstantInputAsIntVector()
391 int index, std::vector<int64_t>* out, xla::ValueInferenceMode mode) { in ConstantInputReshapedToIntVector()
400 xla::ValueInferenceMode mode) { in ConstantInputReshapedToIntVector()
[all …]
|
| D | xla_expression.cc | 155 xla::ValueInferenceMode mode) const { in ResolveConstant()
174 if (mode == xla::ValueInferenceMode::kLowerBound || in ResolveConstant()
175 mode == xla::ValueInferenceMode::kUpperBound || in ResolveConstant()
176 mode == xla::ValueInferenceMode::kValue) { in ResolveConstant()
|
| D | xla_expression.h | 130 xla::ValueInferenceMode mode = xla::ValueInferenceMode::kValue) const;
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | value_inference.h | 69 enum ValueInferenceMode { enum
90 StatusOr<OptionalLiteral> AnalyzeConstant(XlaOp op, ValueInferenceMode mode);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pad_op.cc | 59 xla::ValueInferenceMode::kUpperBound)); in Compile()
118 total_size, xla::ValueInferenceMode::kUpperBound); in Compile()
|
| D | bcast_ops.cc | 50 i, &shape, xla::ValueInferenceMode::kUpperBound)); in Compile()
106 i, &vec, xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| D | conv_ops.cc | 109 xla::ValueInferenceMode::kUpperBound)); in Compile()
177 xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| D | concat_op.cc | 160 1, &inp0_dims, xla::ValueInferenceMode::kUpperBound)); in Compile()
181 1 + i, &inp_dims, xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| D | broadcast_to_op.cc | 35 1, &output_shape, xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| D | sequence_ops.cc | 92 0, &start, xla::ValueInferenceMode::kLowerBound)); in Compile()
94 1, &limit, xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| D | fill_op.cc | 51 "dims", &dims, xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| D | random_ops.cc | 50 0, &shape, xla::ValueInferenceMode::kUpperBound)); in Compile()
127 0, &shape, xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| D | topk_op.cc | 42 xla::ValueInferenceMode::kUpperBound) in Compile()
|
| D | if_while_utils.cc | 49 xla::ValueInferenceMode::kUpperBound); in ConvertCompileTimeConstArgumentsToConst()
|
| D | sparse_to_dense_op.cc | 46 1, &output_shape, xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| D | tile_ops.cc | 70 xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| D | reshape_op.cc | 51 1, &shape_input, xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| D | categorical_op.cc | 48 1, &num_samples, xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| D | segment_reduction_ops.cc | 63 2, &num_segments, xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| D | slice_op.cc | 187 dynamic_size, xla::ValueInferenceMode::kValue); in Compile()
|
| D | tensor_list_ops.cc | 151 1, &num_elements, xla::ValueInferenceMode::kUpperBound)); in Compile()
223 1, &max_num_elements, xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| D | stateless_random_ops_v2.cc | 219 0, &shape, xla::ValueInferenceMode::kUpperBound)); in Compile()
402 0, &shape, xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | segment_reduction_ops.cc | 53 2, &num_segments, xla::ValueInferenceMode::kUpperBound)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | dynamic_shaped_ops.cc | 236 dimension_size, xla::ValueInferenceMode::kUpperBound); in SetDimensionSizeWithRebound()
|