Home
last modified time | relevance | path

Searched refs:shape_representation_fn (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h63 XlaCompiler::ShapeRepresentationFn shape_representation_fn,
72 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn() const { in shape_representation_fn() function
123 XlaCompiler::ShapeRepresentationFn shape_representation_fn; member
Dxla_device_context.h60 XlaCompiler::ShapeRepresentationFn shape_representation_fn,
82 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn() const { in shape_representation_fn() function
Dxla_device.cc142 XlaCompiler::ShapeRepresentationFn shape_representation_fn, in Metadata() argument
147 shape_representation_fn_(std::move(shape_representation_fn)), in Metadata()
197 options.shape_representation_fn, in XlaDevice()
205 shape_representation_fn_(options.shape_representation_fn), in XlaDevice()
Dxla_compile_on_demand_op.cc188 options.shape_representation_fn = metadata.shape_representation_fn(); in Compile()
Dxla_device_context.cc77 XlaCompiler::ShapeRepresentationFn shape_representation_fn, in XlaDeviceContext() argument
85 shape_representation_fn_(std::move(shape_representation_fn)), in XlaDeviceContext()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc163 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn, in BuildComputation() argument
214 if (shape_representation_fn) { in BuildComputation()
217 TF_ASSIGN_OR_RETURN(xla::Shape shape, shape_representation_fn( in BuildComputation()
449 if (!options_.shape_representation_fn) { in XlaCompiler()
450 options_.shape_representation_fn = in XlaCompiler()
628 options_.shape_representation_fn(shape, arg.type)); in XLAShapeForArgument()
646 options_.shape_representation_fn( in XLAShapeForArgument()
1164 options.is_entry_computation ? options_.shape_representation_fn in CompileGraph()
Dxla_compiler.h308 ShapeRepresentationFn shape_representation_fn; member
Dxla_op_kernel.cc391 ctx->compiler()->options().shape_representation_fn( in ReadVariableInputTensor()
530 ctx->compiler()->options().shape_representation_fn(shape, type)); in AssignVariableTensor()
Dxla_compiler_test.cc311 options.shape_representation_fn = in TEST_F()
1034 options.shape_representation_fn = in TEST_F()
1072 options.shape_representation_fn = in TEST_F()
1206 options.shape_representation_fn = in TEST_F()
1276 options.shape_representation_fn = in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc295 options.shape_representation_fn = in CompileToLocalExecutable()
296 platform_info.xla_device_metadata()->shape_representation_fn(); in CompileToLocalExecutable()