Searched refs:CompileTimeConstantInputs (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.h | 188 static Status CompileTimeConstantInputs(const NodeDef& node_def, in CompileTimeConstantInputs() function
191 return CompileTimeConstantInputs(node_def, /*op_kernel=*/nullptr, &op_def, in CompileTimeConstantInputs()
195 static StatusOr<std::vector<int>> CompileTimeConstantInputs( in CompileTimeConstantInputs() function
198 TF_RETURN_IF_ERROR(CompileTimeConstantInputs(node_def, op_def, &out)); in CompileTimeConstantInputs()
206 static Status CompileTimeConstantInputs(const OpKernel& op_kernel, in CompileTimeConstantInputs() function
208 return CompileTimeConstantInputs(op_kernel.def(), /*op_kernel=*/&op_kernel, in CompileTimeConstantInputs()
302 static Status CompileTimeConstantInputs(const NodeDef& node_def,
|
| D | const_analysis.cc | 166 return XlaOpRegistry::CompileTimeConstantInputs(node, *op_def, in GetCompileTimeConstInputs()
169 return XlaOpRegistry::CompileTimeConstantInputs(*op_kernel, in GetCompileTimeConstInputs()
|
| D | mlir_xla_op_kernel.cc | 28 TF_RETURN_IF_ERROR(tensorflow::XlaOpRegistry::CompileTimeConstantInputs( in ContextToXlaArgs()
|
| D | xla_op_registry.cc | 388 /* static */ Status XlaOpRegistry::CompileTimeConstantInputs( in CompileTimeConstantInputs() function in tensorflow::XlaOpRegistry
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | light_outside_compilation.cc | 100 XlaOpRegistry::CompileTimeConstantInputs(node_def, data->op_def)); in CompileToCustomCallCallingTfKernel()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 587 status = tensorflow::XlaOpRegistry::CompileTimeConstantInputs( in LegalizeOp()
|