| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | dynamic_shaped_ops.cc | 232 StatusOr<XlaOp> SetDimensionSizeWithRebound(ValueInference* value_inference, in SetDimensionSizeWithRebound() argument
235 auto inferred_bound_status_or = value_inference->AnalyzeConstant( in SetDimensionSizeWithRebound()
238 auto dynamism_status_or = value_inference->AnalyzeIsDynamic(dimension_size); in SetDimensionSizeWithRebound()
258 StatusOr<XlaOp> SetAllDimensionSizes(ValueInference* value_inference, in SetAllDimensionSizes() argument
260 auto builder = value_inference->builder(); in SetAllDimensionSizes()
270 value_inference->AnalyzeIsDynamic(dim_size)); in SetAllDimensionSizes()
|
| D | dynamic_shaped_ops.h | 45 StatusOr<XlaOp> SetDimensionSizeWithRebound(ValueInference* value_inference,
52 StatusOr<XlaOp> SetAllDimensionSizes(ValueInference* value_inference,
|
| D | BUILD | 161 "//tensorflow/compiler/xla/client:value_inference",
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_expression.cc | 145 xla::ValueInference value_inference(handle().builder()); in ResolveDynamism() local
147 value_inference.AnalyzeIsDynamic(handle())); in ResolveDynamism()
179 xla::ValueInference value_inference(handle().builder()); in ResolveConstant() local
181 value_inference.AnalyzeConstant(handle(), mode)); in ResolveConstant()
|
| D | xla_op_kernel.h | 72 xla::ValueInference& value_inference();
|
| D | BUILD | 441 "//tensorflow/compiler/xla/client:value_inference",
569 "//tensorflow/compiler/xla/client:value_inference",
|
| D | xla_op_kernel.cc | 54 xla::ValueInference& XlaOpKernelContext::value_inference() { in value_inference() function in tensorflow::XlaOpKernelContext
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | BUILD | 214 name = "value_inference",
215 srcs = ["value_inference.cc"],
216 hdrs = ["value_inference.h"],
276 ":value_inference",
|
| D | xla_builder_test.cc | 1441 ValueInference value_inference(&b); in TEST_F() local
1443 value_inference.AnalyzeConstant(out, kUpperBound); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | slice_op.cc | 186 auto constant_size = ctx->value_inference().AnalyzeConstant( in Compile()
199 &ctx->value_inference(), sliced, dynamic_size, i); in Compile()
|
| D | random_ops.cc | 65 SetAllDimensionSizes(&ctx->value_inference(), result, ctx->Input(0)); in Compile()
137 SetAllDimensionSizes(&ctx->value_inference(), result, ctx->Input(0)); in Compile()
|
| D | where_op.cc | 193 &ctx->value_inference(), result, length, 0); in CompileWhereWithSort()
311 &ctx->value_inference(), scattered, num_valid, 0); in CompileWhereWithPrefixSum()
|
| D | pad_op.cc | 117 ctx->value_inference().AnalyzeConstant( in Compile()
|
| D | stateless_random_ops_v2.cc | 258 auto result_or = xla::SetAllDimensionSizes(&ctx->value_inference(), uniform, in Compile()
442 auto result_or = xla::SetAllDimensionSizes(&ctx->value_inference(), normal, in Compile()
|
| D | strided_slice_op.cc | 169 auto lower_bound = ctx->value_inference().AnalyzeConstant( in EmitDynamicSlice()
225 &ctx->value_inference(), slice, in EmitDynamicSlice()
|
| D | BUILD | 215 "//tensorflow/compiler/xla/client:value_inference",
|
| D | image_ops.cc | 602 &context->value_inference(), selected_indices, num_valid, 0); in ComputeResult()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | value_inference_test.cc | 62 ValueInference value_inference(builder); in ComputeDynamismLiteral() local
64 value_inference.AnalyzeIsDynamic(operand)); in ComputeDynamismLiteral()
567 ValueInference value_inference(builder); in ComputeUpperBoundLiteral() local
569 value_inference.AnalyzeConstant( in ComputeUpperBoundLiteral()
730 ValueInference value_inference(builder); in ComputeConstantValueLiteral() local
731 TF_ASSIGN_OR_RETURN(auto literal, value_inference.AnalyzeConstant( in ComputeConstantValueLiteral()
|
| D | BUILD | 2263 "//tensorflow/compiler/xla/client:value_inference",
|