Home
last modified time | relevance | path

Searched defs:sliced_input (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsorting.cc188 XlaOp sliced_input = in TopKWithPartitions() local
224 XlaOp sliced_input = Slice(input, start_indices, limit_indices, strides); in TopKWithPartitions() local
Dslicing.cc32 XlaOp sliced_input = DynamicSlice(input, base_indices, window_sizes); in DynamicStridedSlice() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_resource_ops_test.cc112 ITensorProxyPtr sliced_input = slice_layer->getOutput(0); in NetworkWith2Inputs() local
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dconv_spmd_expander.cc274 mlir::Value sliced_input = builder.create<mlir::TF::SliceOp>( in HandleConv() local